package com.example.mq.mqserver.core;

import com.example.mq.common.ConsumerEnv;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.Data;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * 这个类表示一个消息存储的队列
 */
@Data
public class MSGQueue {

    //队列的名称 (身份标识)
    private String name;
    //表示队列是否持久化
    private boolean durable=false;
    //这个属性如果为ture 表示当前队列只能被一个消费者使用(别人用不了) 为false 则大家都能使用
    //这个独占的功能 后面代码没有实现~
    private boolean exclusive=false;
    //如果当前交换机无人使用 就会自动删除  后续代码并没有实现自动删除功能~
    private boolean autoDelete =false;
    //表示指定的一些额外的参数选项 后续代码并没有实现~!
    private Map<String,Object> arguments=new HashMap<>();
    //当前队列都有那些消费者订阅了
    private List<ConsumerEnv> consumerEnvList =new ArrayList<>();
    //记录当前取到了第几个消费者 方便实现轮询 操作
    private AtomicInteger consumerSeq=new AtomicInteger(0);

    //添加一个新的订阅者
    public void  addConsumerEnv(ConsumerEnv consumerEnv){
            consumerEnvList.add(consumerEnv);
    }
    //订阅者的删除暂不考虑
    //挑选一个订阅者 用来处理当前消息(按照轮询的方式)
    public ConsumerEnv chooseConsumer(){
        if(consumerEnvList.size()==0){
            //该队列没有人订阅
            return null;
        }
        //计算一下当前要取的元素的下标
        int index=consumerSeq.get () %consumerEnvList.size();
        //自增+1
        consumerSeq.getAndIncrement();
        return consumerEnvList.get(index);
    }


    public String getArguments() {
        ObjectMapper objectMapper=new ObjectMapper();
        try {
            //把当前的arguments 从Map 转换成String(JSON)
            return objectMapper.writeValueAsString(arguments);
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
        return "{}";
    }

    public void setArguments(String argumentsJson) throws JsonProcessingException {
        ObjectMapper objectMapper=new ObjectMapper();
        this.arguments = objectMapper.readValue(argumentsJson, new TypeReference<HashMap<String,Object>>(){});
    }

    //为了方便我们在代码中调用arguments 我们再提供一组方法
    public Object getArguments(String key){
        return arguments.get(key);
    }
    public void setArguments(String key,Object value){
        arguments.put(key,value);
    }

    public void setArguments(Map<String,Object> arguments){
        this.arguments=arguments;
    }

}
